package com.xbe.framework.manager;

import com.xbe.framework.shiro.web.session.SpringSessionValidationScheduler;
import javax.annotation.PreDestroy;
import net.sf.ehcache.CacheManager;
import org.apache.shiro.cache.ehcache.EhCacheManager;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

/**
 * 确保应用退出时能关闭后台线程
 *
 * @author liuzhiting
 */
@Component
public class ShutdownManager {
  private static final Logger logger = LoggerFactory.getLogger("sys-user");

  @Autowired(required = false)
  private SpringSessionValidationScheduler springSessionValidationScheduler;

  @Autowired(required = false)
  private EhCacheManager ehCacheManager;

  @PreDestroy
  public void destroy() {
    shutdownSpringSessionValidationScheduler();
    shutdownAsyncManager();
    shutdownEhCacheManager();
  }

  /** 停止Seesion会话检查 */
  private void shutdownSpringSessionValidationScheduler() {
    if (springSessionValidationScheduler != null && springSessionValidationScheduler.isEnabled()) {
      try {
          ShutdownManager.logger.info("====关闭会话验证任务====");
        springSessionValidationScheduler.disableSessionValidation();
      } catch (Exception e) {
          ShutdownManager.logger.error(e.getMessage(), e);
      }
    }
  }

  /** 停止异步执行任务 */
  private void shutdownAsyncManager() {
    try {
        ShutdownManager.logger.info("====关闭后台任务任务线程池====");
      AsyncManager.me().shutdown();
    } catch (Exception e) {
        ShutdownManager.logger.error(e.getMessage(), e);
    }
  }

  private void shutdownEhCacheManager() {
    try {
        ShutdownManager.logger.info("====关闭缓存====");
      if (ehCacheManager != null) {
        CacheManager cacheManager = ehCacheManager.getCacheManager();
        cacheManager.shutdown();
      }
    } catch (Exception e) {
        ShutdownManager.logger.error(e.getMessage(), e);
    }
  }
}
